Correction of lens distortion